package pers.xx.sbm.utils;

import pers.xx.sbm.entity.Menu;

import java.util.ArrayList;
import java.util.List;

/**
 * @author ：xiexing
 * @description：菜单包装类
 * @date ：2020/8/12 13:35
 */
public class MenuUtils {

    /**
     * @description 菜单封装 采用递归方式
     * @author XieXing
     * @date 2020/8/30 15:44
     */
    public static List<Menu> buildMenuTree(List<Menu> menuList, Integer pid) {
        List<Menu> treeList = new ArrayList<>();
        menuList.forEach(menu -> {
            if (pid == menu.getMenuPid()) {
                menu.setChildren(buildMenuTree(menuList, menu.getId()));
                treeList.add(menu);
            }
        });
        return treeList;
    }
}
